//LCR 010. 和为 K 的子数组
/*给定一个整数数组和一个整数 k ，请找到该数组中和为 k 的连续子数组的个数。*/
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0;
        // int hash[2001] = {0};
        unordered_map<int, int> hash;
        hash[0] = 1;
        int ret = 0;
        for (int i = 0; i < nums.size(); i++) {
            sum += nums[i];
            if (hash.count(sum - k))
                ret += hash[sum - k];
            hash[sum]++;
        }
        return ret;
    }
};


//和可被 K 整除的子数组
/*给定一个整数数组 nums 和一个整数 k ，返回其中元素之和可被 k 整除的非空 子数组 的数目。

子数组 是数组中 连续 的部分。*/
class Solution {
public:
    int subarraysDivByK(vector<int>& nums, int k) {
        int sum = 0;
        unordered_map<int, int> hash;
        hash[0] = 1;
        int ret = 0;
        for (int i = 0; i < nums.size(); i++) {
            sum += nums[i];
            if (hash.count((sum % k + k) % k))
                ret += hash[(sum % k + k) % k];
            hash[(sum % k + k) % k]++;
        }
        return ret;
    }
};